#登陆注册
userdb ={}
def register():
    username = input('输入用户名：')
    if username in userdb:
        print('用户以存在，请直接登陆')
    else:
        password = input('请输入密码：')
        userdb[username] =password

def login():
    username = input('用户名：')
    password = input('密码:')
    if userdb.get(username) == password:
        print('\033[32;1m登陆成功\033[0m')
    else:
        print('\033[31;1m登陆失败\033[0m')
def show_menu():
    cmds={'1':register,'2':login}
    prompt = '''
    (1) 注册
    (2) 登陆
    (3) 退出
   请输入选择(1/2/3) :'''
    while 1:
        choice = input(prompt).strip()[0]
        if choice not in '123':
            print('输入错误')
            continue
        if choice==3:
            print('再见')
            break
        cmds[choice]()
if __name__ == '__main__':
    show_menu()